package com.tyrone.algorithm.linear;

import java.util.Iterator;

/**
 * 双向链表
 * @param <T>
 */
public class TwoWayLinkList<T> implements Iterable<T>{
    //记录首节点
    private Node first;
    //记录尾节点
    private Node last;
    //记录链表长度
    private int N;

    public TwoWayLinkList() {
        first=new Node(null,null,null);
        last=null;
        N=0;
    }

    /**
     * 空置线性表
     */
    public void clear() {
        this.first.pre=null;
        this.first.item=null;
        this.last=null;
        this.N=0;
    }

    /**
     * 判断线性表是否为空，是返回true，否返回false
     * @return
     */
    public boolean isEmpty(){
        return N==0;
    }

    /**
     * 获取线性表中元素的个数
     * @return
     */
    public int length(){
        return N;
    }

    /**
     * 读取并返回线性表中的第i个元素的值
     * @param i
     * @return
     */
    public T get(int i){
        Node currNode = first.next;
        for (int j = 0; j < i; j++) {
            currNode=currNode.next;
        }
        return currNode.item;
    }

    /**
     * 向线性表中添加一个元素t
     * @param t
     */
    public void insert(T t){
        if (isEmpty()){
            Node node = new Node(t,null,first );
            last=node;
            first.next=node;
        }else {
            Node oldLast=last;
            Node node = new Node(t,null,oldLast);
            oldLast.next=node;
            last=node;
        }
        N++;
    }


    /**
     * 在线性表的第i个元素之前插入一个值为t的数据元素
     * @param i
     * @param t
     */
    public void insert(int i,T t) {
        if(i<0||i>=N){
            throw new RuntimeException("位置不合法");
        }
        Node pre = first;
        for (int j = 0; j < i; j++) {
            pre=pre.next;
        }
        Node currNode = pre.next;
        Node node = new Node(t,currNode,pre);
        pre.next=node;
        currNode.pre=node;
        N++;

    }


    /**
     * 删除并返回线性表中第i个数据元素
     * @param i
     * @return
     */
    public T remove(int i){
        if (i<0||i>=N){
            throw new RuntimeException("位置不合法");
        }else if(i==N-1){
            Node targetPreNode = first;
            for (int j = 0; j < i; j++) {
                targetPreNode = targetPreNode.next;
            }
            Node targetNode = targetPreNode.next;
            targetPreNode.next=null;
            last=targetPreNode;
            N--;
            return targetNode.item;
        }else{
            Node targetPreNode = first;
            for (int j = 0; j < i; j++) {
                targetPreNode = targetPreNode.next;
            }
            Node targetNode = targetPreNode.next;
            Node targetAftNode = targetNode.next;
            targetPreNode.next = targetAftNode;
            targetAftNode.pre = targetPreNode;
            N--;
            return targetNode.item;
        }
    }

    /**
     * 返回线性表中首次出现的指定的数据元素的位序号，若不存在，则返
     *     回-1
     * @param t
     * @return
     */
    public int indexOf(T t){
        if (t==null){
            throw new RuntimeException("查找的元素不合法");
        }
        Node node =first.next;
        for (int i = 0; node.next!=null; i++) {

            if (node.item.equals(t))
                return i;
            node=node.next;
        }
        return -1;
    }

    /**
     * 获取第一个元素
     * @return
     */
    public T getFirst(){
        if (isEmpty())
            return null;
        return first.next.item;

    }

    /**
     * 获取最后一个元素
     * @return
     */
    public T getLast(){
        if(isEmpty())
            return null;
        return last.item;
    }

    @Override
    public Iterator<T> iterator() {
        return new TIterator();
    }

    private class TIterator implements Iterator{
        private Node node;
        public TIterator(){
            this.node=first;
        }
        @Override
        public boolean hasNext() {
            return node.next!=null;
        }

        @Override
        public Object next() {
            node=node.next;
            return node.item;
        }
    }
    private class Node{
        private  T item;
        private Node next;
        private Node pre;

        public Node(T item, Node next, Node pre) {
            this.item = item;
            this.next = next;
            this.pre = pre;
        }
    }


}
